_base_ = ["./imagenet.py"]
pretrained_model_path = \
    R"D:\workspace\models\MMClassification\dla34-0794-04698d78.pth"
model = dict(type="DLAClassifier",
             pretrained=pretrained_model_path,
             backbone=dict(type="DLA",
                           levels=[1, 2, 2, 1],
                           channels=[64, 128, 256, 512],
                           init_block_channels=32),
             neck=dict(type="GlobalAveragePooling"),
             head=dict(
                 type="LinearClsHead",
                 num_classes=1000,
                 in_channels=512,
                 loss=dict(type="CrossEntropyLoss", loss_weight=1.0),
                 topk=(1, 5),
             ))
